import os
import requests
import re

def remove_think_tags_multiple(text):
    pattern = r'<think>.*?</think>'
    return re.sub(pattern, '', text, flags=re.DOTALL)

# 尝试不同的 URL 格式
urls = [
    #"http://lhf.halphen.cn:21435/api/chat",
    "http://10.42.147.60:11434/api/chat",
]


# 递归获取当前目录及子目录下所有.c文件
c_files = []
for root, dirs, files in os.walk('.'):
    for file in files:
        if file.endswith('.txt'):
            c_files.append(os.path.join(root, file))

for file_path in c_files:
    with open(file_path, 'r', encoding='utf-8') as file:
        file_content = file.read()

    print(f"\n正在处理文件: {file_path}")
    print("="*20+"文件内容"+"="*20)
    print("""分析下面的回复内容是否正确
"""+file_content)

    #赋值一份代码下面用
    code_copy = file_content
    file_content = """阅读分析下面的文章内容，以小见大的分析一下当前经济局势，可能存在的风险和机遇，推导目前利好的资产是什么(现金，股票，黄金，债券，房产等)
""" + file_content
#    file_content += """从以下方面检查这段代码 内存管理(use-after-free, double-free,等) 、指针与数组、逻辑与表达式、函数使用、竞态条件
#    没有明显错误的方面返回无问题，如果存在明显问题请详细解释"""

    data = {
        "model": "deepseek-r1:70b",
        "messages": [
            {"role": "user","content": file_content}],
        "think": True,
        "stream": False
    }

    headers = {
        "Content-Type": "application/json"
    }

    print("="*20+"新闻内容"+"="*20)
    for url in urls:
        try:
            response = requests.post(url, json=data, headers=headers, timeout=10000)
            if response.status_code == 200:
                print(remove_think_tags_multiple(response.json()['message']['content']))
                break
            else:
                print(f"响应内容: {response.text}")
        except Exception as e:
            print(f"错误: {e}")
        print("-" * 50)

